package com.example.demo.leetcode.classics150;

import java.util.Arrays;
import java.util.LinkedList;

/**
 * ******************************************************
 *
 * @author liugh9
 * @version 1.0
 * @classname _50插入区间
 * @description
 * @date 2023/07/24 17:08
 * <p>
 * ******************************************************
 */
public class _50插入区间 {


    /**
     *
     * 给你一个 无重叠的 ，按照区间起始端点排序的区间列表。
     *
     * 在列表中插入一个新的区间，你需要确保列表中的区间仍然有序且不重叠（如果有必要的话，可以合并区间）。
     *
     *
     *
     * 示例 1：
     *
     * 输入：intervals = [[1,3],[6,9]], newInterval = [2,5]
     * 输出：[[1,5],[6,9]]
     * 示例 2：
     *
     * 输入：intervals = [[1,2],[3,5],[6,7],[8,10],[12,16]], newInterval = [4,8]
     * 输出：[[1,2],[3,10],[12,16]]
     * 解释：这是因为新的区间 [4,8] 与 [3,5],[6,7],[8,10] 重叠。
     * 示例 3：
     *
     * 输入：intervals = [], newInterval = [5,7]
     * 输出：[[5,7]]
     * 示例 4：
     *
     * 输入：intervals = [[1,5]], newInterval = [2,3]
     * 输出：[[1,5]]
     * 示例 5：
     *
     * 输入：intervals = [[1,5]], newInterval = [2,7]
     * 输出：[[1,7]]
     *
     *
     * 提示：
     *
     * 0 <= intervals.length <= 104
     * intervals[i].length == 2
     * 0 <= intervals[i][0] <= intervals[i][1] <= 105
     * intervals 根据 intervals[i][0] 按 升序 排列
     * newInterval.length == 2
     * 0 <= newInterval[0] <= newInterval[1] <= 105
     *
     * @param intervals
     * @param newInterval
     * @return
     */
    public int[][] insert(int[][] intervals, int[] newInterval) {
        int[][] temp = new int[intervals.length +1][2];
        for (int i = 0; i < intervals.length; i++) {
            temp[i] = intervals[i];
        }
        temp[intervals.length] = newInterval;
        return merge(temp);
    }


    public int[][] merge(int[][] intervals) {
        Arrays.sort(intervals, (a, b) -> {
            if (a[0] == b[0]) {
                return b[1] - a[1];
            }
            return a[0] - b[0];
        });

        LinkedList<LinkedList<Integer>> res = new LinkedList<>();


        for (int i = 0; i < intervals.length; i++) {
            int[] interval = intervals[i];
            if (i == 0) {
                LinkedList<Integer> first = new LinkedList<>();
                first.addLast(interval[0]);
                first.addLast(interval[1]);
                res.addLast(first);
            } else {
                LinkedList<Integer> last = res.getLast();
                if (interval[0] > last.getLast()) {
                    LinkedList<Integer> item = new LinkedList<>();
                    item.addLast(interval[0]);
                    item.addLast(interval[1]);
                    res.addLast(item);
                } else {
                    if (interval[1] > last.getLast()) {
                        last.removeLast();
                        last.addLast(interval[1]);
                    }
                }
            }
        }


        int[][] result = new int[res.size()][2];
        for (int i = 0; i < res.size(); i++) {
            LinkedList<Integer> curr = res.get(i);
            result[i] = curr.stream().mapToInt(Integer::intValue).toArray();
        }
        return result;
    }
}
